from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())

from openai import OpenAI
client = OpenAI()

import json

system_prompt = """
用户将提供一些示例文本。请解析“问题”和“答案”，并以JSON格式输出。

示例输入:
世界上最高的山是哪座？珠穆朗玛峰。
示例JSON输出:
 {
    "问题": "世界上最高的山是哪座？",
    "答案": "珠穆朗玛峰"
}
"""

user_prompt = "世界上最长的河流是哪一条？尼罗河"

messages = [{"role": "system", "content": system_prompt},
            {"role": "user", "content": user_prompt}]

response = client.chat.completions.create(
    model="deepseek-chat",
    messages=messages,
    response_format={
        'type': 'json_object'
    }
)

print(json.loads(response.choices[0].message.content))